package com.lixw.langchain.enums;

import lombok.AllArgsConstructor;
import lombok.Getter;

/**
 * @ClassName: LLMEnun
 * @description:
 * @author: xuweiLi
 * @create: 2025/8/22 18:49
 **/
@AllArgsConstructor
@Getter
public enum LLMEnum {
    QWEN("qwen-plus","qwenChatModel"),
    DEEPSEEK("deepseek-chat","deepseekChatModel");

    private final String code;

    private final String beanName;

    public static String getLLMEnumByCode(String code) {
        for (LLMEnum e : LLMEnum.values()) {
            if (e.code.equals(code)) {
                return e.beanName;
            }
        }
        throw new IllegalArgumentException("No LLMEnum found for code " + code);
    }
}
